import wx
import wx.adv
import image


class MyTaskBarIcon(wx.adv.TaskBarIcon):
    ID_ABOUT = wx.NewId()  # 菜单选项“关于”的ID
    ID_EXIT = wx.NewId()  # 菜单选项“退出”的ID
    ID_SHOW_Window = wx.NewId()  # 菜单选项“显示页面”的ID
    TITLE = "文件分发工具"  # 鼠标移动到图标上显示的文字

    def __init__(self, window):
        wx.adv.TaskBarIcon.__init__(self)
        self.main_window = window
        self.SetIcon(wx.Icon(image.Logo.getIcon()), self.TITLE)  # 设置图标和标题
        self.Bind(wx.EVT_MENU, self.onAbout, id=self.ID_ABOUT)  # 绑定“关于”选项的点击事件
        self.Bind(wx.EVT_MENU, self.onExit, id=self.ID_EXIT)  # 绑定“退出”选项的点击事件
        self.Bind(wx.EVT_MENU, self.onShowWindow, id=self.ID_SHOW_Window)  # 绑定“显示页面”选项的点击事件
        self.Bind(wx.adv.EVT_TASKBAR_LEFT_DCLICK, self.onShowWindow)

    # “关于”选项的事件处理器
    def onAbout(self, event):
        wx.MessageBox('程序作者：千年墨妖\n最后更新日期：2020-06-28', "关于")

    # “退出”选项的事件处理器
    def onExit(self, event):
        self.main_window.Close()
        wx.Exit()

    # “显示页面”选项的事件处理器
    def onShowWindow(self, event):
        self.main_window.Show()

    # 创建菜单选项
    def CreatePopupMenu(self):
        menu = wx.Menu()
        for mentAttr in self.getMenuAttrs():
            menu.Append(mentAttr[1], mentAttr[0])
        return menu

    # 获取菜单的属性元组
    def getMenuAttrs(self):
        return [('显示主窗口', self.ID_SHOW_Window),
                ('关于', self.ID_ABOUT),
                ('退出', self.ID_EXIT)]


class MyFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self)
        MyTaskBarIcon(self)  # 显示系统托盘图标


class MyApp(wx.App):
    def OnInit(self):
        MyFrame()
        return True


if __name__ == "__main__":
    app = MyApp()
    app.MainLoop()
